# encoding: utf-8
# @Time:    :2024/12/22 15:26
# @Description: 常用变量定义

import torch

# 待训练大模型，这里使用Qwen2.5 0.5b作为demo
pretrained_model = "E:/codes/nlp_about/pretrained_model/Qwen2_5-0_5b"
# 文本分类模型，使用roberta-chinese
roberta_pretrained_model = "E:/codes/nlp_about/pretrained_model/hfl_chinese-roberta-wwm-ext"
# device
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")


train_data_path = r"E:/codes/nlp_about/reinforcement-learning-demos/datas/data/train.json"
test_data_path = r"E:/codes/nlp_about/reinforcement-learning-demos/datas/data/test.json"
